import pymysql
import pinyin
from settings import DB_HOST, DB_NAME, DB_PORT, DB_USER, DB_PASSWORD


def delete_user(name, position):
    global cur, conn
    position = position.replace('组', '')
    if position not in ['后端', '前端', '测试', '中台']:
        return "Other"
    username = pinyin.get(name, format='strip')
    try:
        conn = pymysql.connect(host=DB_HOST,
                               user=DB_USER,
                               password=DB_PASSWORD,
                               port=DB_PORT,
                               db=DB_NAME
                               )
        cur = conn.cursor()
        cur.execute('DELETE FROM `users_user` WHERE `username`="%s"' % username)
        conn.commit()
        return "%s已从Jumpserver中删除" % name
    except Exception as e:
        return e
    finally:
        cur.close()
        conn.close()
